package com.example.salesystem.model.pageModel;

import cn.hutool.core.bean.BeanUtil;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiModelProperty;
import lombok.Data;

import java.util.Collections;
import java.util.List;
import java.util.function.Function;
import java.util.stream.Collectors;

@Data
@ApiModel(description = "分页结果")
public class PageResult<T> {
    @ApiModelProperty("总条数")
    private Long total;
    @ApiModelProperty("总页数")
    private Long pages;
    @ApiModelProperty("结果集合")
    private List<T> list;

    /**
     * 这里的 PO 指的是数据表实体类对象
     * 这里的 VO 指的是返回值实体类对象
     * @param p
     * @param <PO>
     * @param <VO>
     * @return
     */
    public static <PO,VO> PageResult<VO> of(Page<PO> p,Class<VO> clazz) {
        PageResult<VO> res = new PageResult<>();
        // 1.总条数
        res.setTotal(p.getTotal());
        // 2.总页数
        res.setPages(p.getPages());
        // 3.当前页数据
        List<PO> records = p.getRecords();
        if (records == null || records.isEmpty()) {
            res.setList(Collections.emptyList());
            return res;
        }
        // 4.拷贝（类型转换）
        res.setList(BeanUtil.copyToList(records,clazz));
        // 5.返回
        return res;
    }

    public static <PO,VO> PageResult<VO> of(Page<PO> p, Function<PO,VO> convertor) {
        PageResult<VO> res = new PageResult<>();
        // 1.总条数
        res.setTotal(p.getTotal());
        // 2.总页数
        res.setPages(p.getPages());
        // 3.当前页数据
        List<PO> records = p.getRecords();
        if (records == null || records.isEmpty()) {
            res.setList(Collections.emptyList());
            return res;
        }
        // 4.拷贝（类型转换）
        res.setList(records.stream().map(convertor).collect(Collectors.toList()));
        // 5.返回
        return res;
    }

    public static <T> PageResult<T> of(Page<T> p) {
        PageResult<T> res = new PageResult<>();
        // 1.总条数
        res.setTotal(p.getTotal());
        // 2.总页数
        res.setPages(p.getPages());
        // 3.当前页数据
        List<T> records = p.getRecords();
        if (records == null || records.isEmpty()) {
            res.setList(Collections.emptyList());
            return res;
        }
        // 4.拷贝
        res.setList(records);
        // 5.返回
        return res;
    }
}
